# Configuration file for the Sphinx documentation builder.
#
# This file only contains a selection of the most common options. For a full
# list see the documentation:
# https://www.sphinx-doc.org/en/master/usage/configuration.html

# -- Path setup --------------------------------------------------------------

# If extensions (or modules to document with autodoc) are in another directory,
# add these directories to sys.path here. If the directory is relative to the
# documentation root, use os.path.abspath to make it absolute, like shown here.
#
# import os
# import sys
# sys.path.insert(0, os.path.abspath('.'))

# -- Project information -----------------------------------------------------
import glob
import os
import re
import shutil

import sphinx_gallery.gen_rst

import minari


project = "Minari"
copyright = "2022"
author = "Farama Foundation"

# The full version, including alpha/beta/rc tags
release = minari.__version__


# -- General configuration ---------------------------------------------------

# Add any Sphinx extension module names here, as strings. They can be
# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
# ones.
extensions = [
    "sphinx.ext.napoleon",
    "sphinx.ext.doctest",
    "sphinx.ext.autodoc",
    "sphinx.ext.githubpages",
    "sphinx.ext.viewcode",
    "furo.gen_tutorials",
    "myst_parser",
    "sphinx_gallery.gen_gallery",
    "sphinx_github_changelog",
]

# Add any paths that contain templates here, relative to this directory.
templates_path = ["_templates"]

# List of patterns, relative to source directory, that match files and
# directories to ignore when looking for source files.
# This pattern also affects html_static_path and html_extra_path.
exclude_patterns = ["tutorials/README.rst"]

# Napoleon settings
napoleon_use_ivar = True
napoleon_use_admonition_for_references = True
# See https://github.com/sphinx-doc/sphinx/issues/9119
napoleon_custom_sections = [("Returns", "params_style")]

# Autodoc
autoclass_content = "both"
autodoc_preserve_defaults = True

# -- Options for HTML output -------------------------------------------------

# The theme to use for HTML and HTML Help pages.  See the documentation for
# a list of builtin themes.
#
html_theme = "furo"
html_title = "Minari Documentation"
html_baseurl = "https://minari.farama.org"
html_copy_source = False
html_favicon = "_static/img/favicon.png"
html_theme_options = {
    "light_logo": "img/Minari.svg",
    "dark_logo": "img/Minari_White.svg",
    "image": "img/robotics-github.png",
    "description": "Minari is a Python library for conducting research in offline reinforcement learning.",
    "gtag": "G-R5TRTT6R78",
    "versioning": True,
    "source_repository": "https://github.com/Farama-Foundation/Minari/",
    "source_branch": "main",
    "source_directory": "docs/",
}

html_static_path = ["_static"]
html_css_files = ["css/termynal.css", "css/custom.css", "css/directory_tree.css"]
html_js_files = ["js/termynal.js", "js/custom.js"]

# -- Generate Tutorials -------------------------------------------------

sphinx_gallery.gen_rst.EXAMPLE_HEADER = """
.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "{0}"
.. LINE NUMBERS ARE GIVEN BELOW.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_{1}:

"""

sphinx_gallery_conf = {
    "ignore_pattern": r"__init__\.py",
    "examples_dirs": "./tutorials",
    "gallery_dirs": "./tutorials",
    "show_signature": False,
    "show_memory": False,
    "min_reported_time": float("inf"),
    "filename_pattern": f"{re.escape(os.sep)}run_",
    "default_thumb_file": os.path.join(
        os.path.dirname(__file__), "_static/img/minari-text.png"
    ),
}

# -- Generate Changelog -------------------------------------------------

sphinx_github_changelog_token = os.environ.get("SPHINX_GITHUB_CHANGELOG_TOKEN")


def move_404(app, _):
    src = app.outdir.joinpath("404/index.html")
    dst = app.outdir.joinpath("404.html")
    shutil.move(src, dst)

    with open(dst, "r+") as fp:
        content = fp.read()
        content = content.replace('href="../', 'href="/').replace('src="../', 'src="/')
        fp.seek(0)
        fp.truncate()
        fp.write(content)


def move_dataset_gifs(app, _):
    gif_pathname = app.srcdir.joinpath("datasets", "**", "*.gif")
    for file in list(glob.glob(str(gif_pathname), recursive=True)):
        dest_path = app.outdir + file[len(app.srcdir) :]
        shutil.copy(file, dest_path)


def setup(app):
    app.connect("build-finished", move_404)
    app.connect("build-finished", move_dataset_gifs)
